Added comments in code and started writing the README file