Clean up readme & comments