Added comments and making the README better