2.0.1

-UI fixes