First attempt on hw 1