Added some end to end testing