Cleaned up a exercise4 code a bit