solved small amount of problem