Change functionality of the code