Customizing matplotlib with style