updating comments and fixing code