Button code cleaning.

cleaning