Updated a little. Fix styles