Added New code from Udemy Blog