Minor code clean up in models