Added comments + changed the README file