1.add database to record stages
2.add return button
3.add button to choose stages