Fixing sintaxis
Fixing importing