Added missing files - little fixes