Adds some model tests