Added some tests. 1 failing