adding assignment 4 on unit 3