updated tests
added mock data