Adding withdrawal function in the model