Tests added

Thanks :)