Added more of his code. Need to still revise