fixed Assignment two, made mistake with .