added assignment 2; but unsure if correct