Added tests coverage