second version
extra credit done