Add needed methods