Added comments in code and updated README