Improved readme and comments
improved comments