Slight better code and readme