Updated code :repeat: