Assignment 4 one shot