Correct Output

still need to clean up the code little bit