Updates to withdrawal function