Added some end to end tests