Cleaning up code
Added lines output and things are looking good