Adding simple test to pass all