Add class and change GUI a little bit.