Updated code
removed puma