Add function to obtain drift potential