bugfix

using correct arguments
cleaning the code

removing unneeded cats and prints