Changed method signature

Added proper arguments.