Add missing imports causing exceptions when trying to run