Removing unnecessary datetime import