Add code

Wrote the basic skeleton