BUG: Make it backwards compatible
BLD: Release v0.4.4