- minor fixing and removed unnecessary import