Change solution slightly