Add UI classes