Assignment 3b addition fix