Trying to get the code that works