Few changes to make sure I was calling the correct method names