UI tweaks and updates. Pushing to release 0.2.