made some changes to solution 4