Add a better solution to gas-station.cpp