Code done

....hopefully