Refactor duplicated code
Pull button, add exception for initial = 0
Add frames