Fix code styling

Add forgotten rules