Add completed code for reference