Add cont, fix bug