Trying to write a clean code