added comments, cleaned up interface, fixed colors on plots