Adding some end-to-end tests