First try at assignment 2