fixed bug, added styling