model bug fix
minor styling fix