Add more modes to `axt`