added testing and got first test to pass