updated code with references