Fix imports
Take array as input