Adds UI tests

Refs #25