Added tests
Adding missed file from last commit. ^^;;