More pythonic way of creating dict