add styles / bug fix