Cleaned up code and added comments for submission for week 1-java2