Added testing code