High Quality Code Homework - Fixes