Research Assignment code working