Improved readme and comments