Added more comments

Tried to make it more understandable what's going on