Added tests which passes