Fixed style a little bit