Made some little tweaks to models