finalize styling + add button for restart