Added import and fixed name issue