Add more styling features, refactor the Style method