Updated the readme and restructured the package