More comments in code

Take lines from the readme to explain what is happening