Shape button working again