better code nice python