Fixed some errors in the code

Fixed the location of the button in each situation
Fixed the colors from changing weirdly