Refactor

added function