Made changes according to teacher notes. Much better result now!