DOC: Bump version to 0.3.1
RF: Minor cleanup