my contribution

Changed method to set variable.