cleaned variable names
removed unnecessary import