Added functionality passed test