Python halo mass function calculator