Homework 1 - frist working version