fixed small bug and did a little styling