First attempt assignment 3