added styling and fixed bug