Made the expansion button responsive