Add required imports