UI fix 

Pulled down because I did not want to wait for travis. This should just be UI changes, everything looks good so made Pr & merging for ya @AdamBirdsall.

```
6 scenarios (6 passed)
34 steps (34 passed)
```