cleaned up comments and added new cout