Fixed styling accordingly

Made a couple changes to make styling work right.