Add some comments
Clean up a little
Try colors