minor update, removed test class