Fixed missing imports and container for method.