Fix comments in header
Home Work 2

First attempt