Code cleanup

added project class header