Final code that was turned in