Making corrections

Made some minor corrections to my code