A slightly simpler solution