Adding Pranali code from assignment 6