Небольшие исправления в коде