simple visualization of molecules in python