Updated KAT demo, fixes #4