Added basic tests :)