Writing more tests for views and logic