Final changes to A1 assignment after testing.