Added comments and changed button size