Add UI for testing