CSE502 code added