Fixed up code/added comments