Add more code to test