Added loss value as a parameter
Updated readme