Made code prettier

Deleted unnec stuff, added some comments