Added missing files to solution