Made some comments and fixed a stupid error on my part