finished assignment :smirk: fixed #4