Tidying up, add button to timeline