Add main gameplay logic