Assignment 4: Fix minor warnings