add code

draft of code