make it calculate correct