cleaned up a bit
better commenting