Updated Code CSE 02