unit4 update 5

rearranging code