Updated the cousework - simple but working!