added my test code