Quick and dirty update.
Added missing files.