Added better model