better code in readme