Yeah, I wanted to do that but due to my inexperience and lack of time I couldn't. So, I instead relied on hacky solution of printing message. I will try to update it in future though.
