functionally working. Adding style