added files and small improvement