Adding the code for petrol pump problem