Please help me make this solution better