Add more tests for the backend