Code after submitting assingment