better code style on the model