Added missing import and updated README