Updated code to video 26