Button class update
I DIDN'T EVEN IMPORT SWING IN MAINSCREEN :D