Add sample unit/funtional test