Fix code style

This is a mess.