Changed arguments format and removed an unnecessary import