Adding slution to hmework 3 week 3