Update Stian-assign4

updated assignment 4