Add type as prop to loading button