Starter code for homework
First test is done
Second test