Make module Robust enough to take types of inputs

[adit4443ya]

Reference :-#61 (comment)

TODO:

• Make changes in various subroutines to have different datatypes of inputs.

We will append this TODO with specific details once all C-wrappers are removed