Adding simple functional tests