Change output to method