Task2 - Fixed model, now GUI plz