Update square withdrawal to have all the behaviors