fixed small bug and did a little styling

@datamadic 
there's other things that need to be done still here....