Fixed issue in btn class to being set correctly