Add back missing files.. no idea what i'm doing :)