updated code

still working on my code.