Minor solution edits