Refactor css setup