Edited README and added a few comments