use dict comprehention