Add actual tests