Fixed the small changes made