Kinda finished homework 4