Remove unnecessary code (test function)