Cleaned spacing for the code sections

Cleaned up spacing for code sections and added for author tag in front of TODO. Not sure if the TODO is needed.