Added my changed code.