Made changes to the code based on Joe's comments