Pretty much done with first assigment