now code looks a bit better