Add simple testing