Closes #4: Added first tests