My code from following the walktrough on TeamTreeHouse