improved css and fixed bug