UKG interview question

Fix some code that involved OOP concepts.