Add some comments and clean up a little