Add inference + testing