Adding missing import and variable