Working version

State of code before interview