Added functionality
Updated readme