Missing import statement added