Adding code for two different parts of the project, because bad practice
is awesome