Correct errors in header
Commit solution to problem 4