made changes to solution