Adding initial logic and styling