Removed unused import
Added missing future imports