Small updates to our compiler and Python API