add code comments and cleanup