Replace while loop with if loop and button click. Need to work on CSS and button posistioning relative to section and p