Add unti tests