Add peewee support