Edited comments, cleaned up code.