added some styles to the refactored version