Add some optimizazion