Tidy-up: Adding of header comments etc. throughout project and general code clean up.