Added comments

Might add an if statement later on