Fixed import errors running main script.