cleaned up comments on the code