Added styling functionality