Added test to solution