Fix classes
Update style "code" screen
Update style