Wrote comments in code, did a little clean up