Improved the readme and added features