Changed a few comments and removed unecessary lines in test