Add code

I rethought #2 and think this should have some code.