Add code directory, copied over from PA6