removing test code
fixing code