Added some extra comments and modified the readme