Updated challenge4

Fixed a few minor errors and cleaned up a few things.

Signed-off-by: Jonathan Tomek <jonathan.tomek@gmail.com>