Clean up imports in Logic.java