added functionality to logic