Fixed the readme + small fixes