Adding test for working?