minor solution edits