added tests.. currently needs to be run test folder