Add tests for viewing and removing habits