Improve code

[sameerkumar18]

I wrote this code about 4 months ago when I was trying to implement REST in Flask API.
I think the code can be improved a lot.

[esouthren]

I'll have a bash at this

Pull Request

[brunoerg]

I'll help you!

[aspittel]

See #14!

[earvingemenez]

I opened a PR for this #28

[Nitish18]

Hi sameer,
take a look at this PR #31