minor solution changes