Edited controller to adjust with new command methods