Added tests and improving stuff