Seperated module into 2 functions, instead of relying on arguament to function