Started writing tests for some UI