Added first tests :D