Made method public and fixed return