lab 4 improves - add proper messages
Start of lab5