Change hardcoded values to use variable