adjust demo with refactoring og AG lib