Updated capstone project