Update styling

@alex-deaconu please review