Remove babel and add pyyaml