added suggested solution