Add block recognition and correct style