refactor, optimization and error fix

nekoj mal refaktoring poradi kvalitet na kod, mala optimizacija kaj prebaruvanje na grades i mislam deka e sredena greshkata so grades