1 files changed, 2 insertions, 4 deletions
diff --git a/src/ui.rs b/src/ui.rs
index dd3bfc5..3307123 100644
--- a/src/ui.rs
+++ b/src/ui.rs
@@ -13,12 +13,10 @@ use tui::{
     Frame, Terminal,
 };
 
-use crate::Calculator;
+use crate::{Calculator, Function};
 
 const WIDTH: u16 = 32;
 
-const FUNCTIONS: [&str; 3] = ["sin", "cos", "tan"];
-
 impl Calculator {
     pub fn draw<B: Backend>(&self, f: &mut Frame<B>) {
         let chunks = layout(self.stack.len(), f.size());
@@ -85,7 +83,7 @@ fn input_buffer_widget(input_buffer: &str) -> impl Widget {
         Span::raw(">"),
         Span::styled(
             format!(" {}", input_buffer),
-            if FUNCTIONS.contains(&input_buffer) {
+            if let Ok(_) = Function::parse(&input_buffer) {
                 Style::default().add_modifier(Modifier::BOLD)
             } else {
                 Style::default()