My little additions

Just downloaded your fork and had some issues, so here are some small fixes :)