Cleaned out unused import in favour of class function