Add context and imports