added in missing import statements and library statement