adding in assignment 5b