Removed unnecessary code, fixed running command, updated readme.