adding functionality tests