Enhanced Distribution Modelling via Augmented Architectures For Neural ODE Flows