Added basic testing