updated the code with schema to make it more generic
Merge pull request #4 from pkpramod005/master

updated the code with schema to make it more generic