Updated code from review