Code is fixed.

Are you allowed to use functions besides just main() ?