Took out unnecessary input and added comments