Refactored the "follow button"