Add capture code to game