added styling and removed prints