Couple of minor changes to fix the css for now, I'll work on it more later