Removed the comments and the prints, also removed the unnecessary test