v 2.0.1

Some minor changes