Clean up README + comments