Clean up of solution
minor change