Adding button to keep the build open