Fix complex code?
Add some style