Missing import and some indentation