Added end to end test